import torch #必须先导入torch
import ops_test_op


device='cuda:0'
input1 = torch.tensor([[0,1,2],[3,4,5]]).to(torch.float32).to(device)
input2 = torch.tensor([[100,100,100],[100,100,100]]).to(torch.float32).to(device)
output = torch.tensor([[0.0,0.0,0.0],[0.0,0.0,0.0]]).to(torch.float32).to(device)

ops_test_op.test(input1,input2,output)
print(output)
pass